package com.lzj.demo.collection.student;

import java.util.ArrayList;
import java.util.Scanner;

public class Student01Manger {
    public static void main(String[] args) {

        ArrayList<Student01> array = new ArrayList<Student01>();
        while (true) {
            System.out.println("------欢迎来到学生管理系统------");
            System.out.println("1 添加学生");
            System.out.println("2 修改学生");
            System.out.println("3 删除学生");
            System.out.println("4 查看学生信息");
            System.out.println("5 退出");
            Scanner sc = new Scanner(System.in);
            String line = sc.nextLine();
            switch (line) {
                case "1":
                    // System.out.println("1 添加学生");
                    addStudent01(array);
                    break;
                case "2":
                    //System.out.println("2 修改学生");
                    updateStudent01(array);
                    break;
                case "3":
                    //System.out.println("3 删除学生");
                    deleteStudent01(array);
                    break;
                case "4":
                    // System.out.println("4 查看学生信息");
                    findAllStudent01(array);
                    break;

                case "5":
                    System.out.println("5 退出");
                    System.exit(0);
            }
        }

    }

    public static void addStudent01(ArrayList<Student01> array) {
        Scanner sc = new Scanner(System.in);
        String aid;
        while (true) {
            System.out.println("请输入学生学号");
            aid = sc.nextLine();
            boolean flag = isUsed(array, aid);
            if (flag) {
                System.out.println("你输入的账号已经存在，请重新输入");
            } else {
                break;
            }
        }
        System.out.println("请输入学生姓名");
        String name = sc.nextLine();
        System.out.println("请输入学生年龄");
        String age = sc.nextLine();
        System.out.println("请输入学生住址");
        String address = sc.nextLine();

        Student01 s = new Student01();
        s.setAid(aid);
        s.setName(name);
        s.setAge(age);
        s.setAddresss(address);
        array.add(s);

    }

    public static boolean isUsed(ArrayList<Student01> array, String aid) {
        boolean flag = false;
        for (int i = 0; i < array.size(); i++) {
            Student01 s = array.get(i);
            if (s.getAid().equals(aid)) {
                flag = true;
                break;
            }
        }
        return flag;
    }

    public static void findAllStudent01(ArrayList<Student01> array) {
        if (array.size() == 0) {
            System.out.println("无信息，请添加信息后再查询");
            return;
        }

        System.out.println("学号\t\t\t姓名\t\t年龄\t\t地址");
        for (int i = 0; i < array.size(); i++) {
            Student01 s = array.get(i);
            System.out.println(s.getAid() + "\t" + s.getName() + "\t" + s.getAge() + "岁\t\t" + s.getAddresss());
        }
    }

    public static void deleteStudent01(ArrayList<Student01> array) {
        Scanner sc = new Scanner(System.in);
        System.out.println("请输入要删除学生学号");
        String aid = sc.nextLine();
        int index = -1;
        for (int i = 0; i < array.size(); i++) {
            Student01 s = array.get(i);
            if (s.getAid().equals(aid)) {

                index = i;
                break;
            }
        }
        if (index == -1) {
            System.out.println("你输入的学号不存在，请重新输入");
        } else {
            array.remove(index);
            System.out.println("删除学生成功");
        }
    }

    public static void updateStudent01(ArrayList<Student01> array) {
        Scanner sc = new Scanner(System.in);
        System.out.println("请输入你要修改的学生的学号");
        String aid = sc.nextLine();
        int index = -1;
        {
            for (int i = 0; i < array.size(); i++) {
                Student01 student01 = array.get(i);
                if (student01.getAid().equals(aid)) {
                    index = i;
                    break;
                }
            }
            if (index == -1) {
                System.out.println("你输入的学号不存在，请重新输入");
            }
            else {
        System.out.println("请输入学生的新姓名");
        String name = sc.nextLine();
        System.out.println("请输入学生的新年龄");
        String age = sc.nextLine();
        System.out.println("请输入学生的新地址");
        String address = sc.nextLine();
        Student01 s = new Student01();
        s.setAid(aid);
        s.setName(name);
        s.setAge(age);
        s.setAddresss(address);



                array.set(index, s);
                System.out.println("修改学生成功");

            }

        }
    }
}